#coding:utf-8
import pickle
def register():
    file1 = open("data.txt","rb")
    the_data_base = pickle.load(file1)

#判断用户名是否已经存在
    while True:   #用户名是否已经存在
        reg_name = input("请输入你要注册的用户名：")
        if reg_name not in the_data_base.keys():
            break
        else:
            print("用户名以存在请重新取名！")
            continue

#判断密码是否两次一致
    while True:
        reg_passwd = input("请设置你的密码：")
        reg_passwd_copy = input("请再次确定你的密码：")
        if reg_passwd == reg_passwd_copy:
            break
        else:
            print("两次密码不一致，请重新输入！")

#把用户名和密码写入data.txt文件
    the_data_base[reg_name] = reg_passwd  #用字典的方式赋值
    file1.close()
    file2 =open("data.txt","wb")
    pickle.dump(the_data_base,file2)
    file2.close()


